package com.niit.mobileDevide.userbehaviorclass;

import org.apache.hadoop.io.Text;
import org.apache.hadoop.mapreduce.Partitioner;

public class DevideUserbehaviorClassPartitioner extends Partitioner<Text, Text> {

    @Override
    public int getPartition(Text key, Text value, int numPartitions) {
        // 根据用户行为分类进行分区
        String userBehaviorClass = key.toString().toLowerCase();
        if ("1".equals(userBehaviorClass)) {
            return 0; // 轻度用户分配到第一个分区
        } else if ("2".equals(userBehaviorClass)) {
            return 1; // 重度用户分配到第二个分区
        } else if ("3".equals(userBehaviorClass)) {
            return 2; // 重度用户分配到第二个分区
        }else if ("4".equals(userBehaviorClass)) {
            return 3; // 重度用户分配到第二个分区
        }else if ("5".equals(userBehaviorClass)) {
            return 4; // 重度用户分配到第二个分区
        }else {
            // 对于无效的用户行为分类，分配到默认分区0
            return 0;
        }
    }
}